from utils.registry import Registry
from monai.losses import DiceLoss

# 损失函数注册器
LOSSES = Registry('loss')

@LOSSES.register()
class DiceLossWrapper(DiceLoss):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)